package javadesign.creational.factory;

import static org.junit.Assert.*;

import org.junit.Test;

public class MilkShakeFactoryTest {

	@Test
	public final void testGetChocolateMilkShake() {
		MilkShake  milkShake =MilkShakeFactory.getMilkShake(1);
		if(milkShake instanceof ChocolateMilkShake){
		 assertTrue(true);	
		}else{
			fail("Chocolate Milkshake not returned");
			
		}
	}
	
	@Test
	public final void testGetButterScotchMilkShake() {
		MilkShake  milkShake =MilkShakeFactory.getMilkShake(2);
		if(milkShake instanceof ButterScotchMilkShake){
		 assertTrue(true);	
		}else{
			fail("ButterScotch Milkshake not returned");
		}
		
	}

}
